package cn.hfatec.shms.web.yk.ykSr.service;

import cn.hfatec.Page;
import cn.hfatec.Table;
import cn.hfatec.base.BaseDao;
import cn.hfatec.shms.cache.CodeInfoCache;
import cn.hfatec.shms.enums.CodeType;
import cn.hfatec.shms.system.codeInfo.model.CodeInfo;
import cn.hfatec.shms.web.yk.ykSr.dao.YkSrDao;
import cn.hfatec.shms.web.yk.ykSr.model.YkSr;
import cn.hfatec.shms.web.yk.ykSr.model.YkSrParams;
import org.apache.poi.hssf.usermodel.*;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;


/**
 *
 * @author hfatec
 * @version 1.0
 *
 * the JAVA code is generate by middlegen
 *
 */
@Service("YkSrService")
public class YkSrService {

    public static final String BEAN_ID = "YkSrService";

    //注入Dao层
    @Resource(name = "YkSrDao")
    private YkSrDao ykSrDao;

    //注入BaseDao
    @Resource(name = "BaseDao")
    private BaseDao baseDao;

    @Transactional
    public static String encodeFilename(HttpServletRequest request, String fileName) {
        String rtn = "";
        try {
            String userAgent = request.getHeader("USER-AGENT");
            String new_filename = URLEncoder.encode(fileName, "UTF8");

            rtn = "filename=\"" + new_filename + "\"";

            if (userAgent != null) {
                userAgent = userAgent.toLowerCase();

                if (userAgent.indexOf("msie") != -1) {
                    rtn = "filename=\"" + new_filename + "\"";
                } else if (userAgent.indexOf("opera") != -1) {
                    rtn = "filename*=UTF-8''" + new_filename;
                } else if (userAgent.indexOf("safari") != -1) {
                    rtn = "filename=\"" + new String(fileName.getBytes("UTF-8"), "ISO8859-1") + "\"";
                } else if (userAgent.indexOf("applewebkit") != -1) {
                    new_filename = new String(fileName.getBytes("UTF-8"), "ISO-8859-1");
                    rtn = "filename=\"" + new_filename + "\"";
                } else if (userAgent.indexOf("mozilla") != -1) {
                    rtn = "filename*=UTF-8''" + new_filename;
                }
            }
        } catch (UnsupportedEncodingException e) {
            try {
                rtn = new String(fileName.getBytes("UTF-8"), "iso-8859-1");
            } catch (UnsupportedEncodingException localUnsupportedEncodingException1) {
            }
        }
        return rtn;
    }

    /**
     *
     * 查询数据
     *
     * @param ykSrParams
     * @param page
     *
     */
    @Transactional
    public Table queryYkSrTable(YkSrParams ykSrParams, Page page){

        HashMap<String, Object> params = new HashMap<String, Object>();

        String bzStart = null;
        if ((ykSrParams.getBillTime() == null) || (ykSrParams.getBillTime().isEmpty())) {
            bzStart = getBzstart();
            ykSrParams.setBillTime(bzStart);
        } else {
            bzStart = ykSrParams.getBillTime();
        }

        if ((ykSrParams.getBillEnd() == null) || (ykSrParams.getBillEnd().isEmpty())) {
            String  riqiend = getBzend(bzStart);
            ykSrParams.setBillEnd(riqiend);
        }
        params.put("params", ykSrParams);
        return baseDao.queryTableByPage("queryYkSr", params, page);
    }

    public String getBzstart() {
        Calendar cal = Calendar.getInstance();
        SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
        cal.set(Calendar.DAY_OF_WEEK, Calendar.MONDAY);
        cal.set(Calendar.DAY_OF_WEEK, Calendar.SUNDAY);
        String bzstart = df.format(cal.getTime());
        return bzstart;
    }

    public String getBzend(String bzStart) {
        SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
        Date myDate = null;
        String bzend = null;
        try {
            myDate = formatter.parse(bzStart);
            Calendar c = Calendar.getInstance();
            c.setTime(myDate);
            c.add(Calendar.DATE, 6);
            myDate = c.getTime();
            bzend = formatter.format(myDate);
        } catch (ParseException e) {
            e.printStackTrace();
        }
        return bzend;
    }

    /**
     *
     * 插入数据
     * @param ykSr
     *
     */
    @Transactional
    public void saveYkSr(YkSr ykSr) {
        if(ykSr.getId() != null) {
            YkSr result = getYkSr(ykSr.getId());
            if(result != null) {
                ykSrDao.update(ykSr);
            } else {
                ykSrDao.insert(ykSr);
            }
        } else {
            ykSrDao.insert(ykSr);
        }
    }

    /**
     *
     * 删除数据
     * @param id
     *
     */
    @Transactional
    public void deleteYkSr(Integer id) {

        ykSrDao.delete(id);
    }

    /**
     *
     * 删除数据
     * @param ykSrList
     *
     */
    @Transactional
    public void deleteYkSrBatch(List<String> ykSrList) {
        for(String id : ykSrList) {
            deleteYkSr(Integer.valueOf(id));
        }
    }

    /**
     *
     * 根据ID获取数据
     * @param id
     *
     */
    @Transactional
    public YkSr getYkSr(Integer id) {

        return ykSrDao.getYkSrById(id);
    }

    /**
     * 导出
     *
     * @param
     */
    @Transactional
    public String exportExcel(YkSrParams ykSrParams, HttpServletRequest request, HttpServletResponse response) {
        try {
            String clientFileName = encodeFilename(request, System.currentTimeMillis() + "");
            response.reset();
            response.setHeader("Content-Disposition", "attachment; " + clientFileName + ".xls");
            response.setContentType("application/octet-stream;charset=UTF-8");
            OutputStream os = response.getOutputStream();
            List dataList = ykSrDao.queryYkSr(ykSrParams);
            List nameList = new ArrayList();

            nameList.add("项目");
            nameList.add("小项");
            nameList.add("三师1组");
            nameList.add("三师2组");
            nameList.add("客服组");
            nameList.add("三师3组");
            nameList.add("拓展部");
            nameList.add("本部");
            nameList.add("合计");

            this.exportExcel(nameList, dataList, os);
        } catch (Exception e) {
            e.printStackTrace();
        }

        return "";
    }

    /**
     * @return HSSFWorkbook
     */
    public HSSFWorkbook createWorkbook(List fieldName, List fieldData) {
        List<CodeInfo> sexList = CodeInfoCache.getCodeInfo(CodeType.SEX);
        // 第一步，创建一个webbook，对应一个Excel文件
        HSSFWorkbook wb = new HSSFWorkbook();
        // 第二步，在webbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet("营业收入");
        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
        HSSFRow row = sheet.createRow((int) 0);
        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        HSSFCellStyle cellstyle = (HSSFCellStyle) wb.createCellStyle();// 设置表头样式
        cellstyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER); // 设置垂直居中
        cellstyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);// 设置水平居中
        cellstyle.setWrapText(true); // 设置为自动换行
        HSSFFont cellFont = (HSSFFont) wb.createFont(); // 创建表头字体样式
        cellFont.setFontHeightInPoints((short) 10); // 设置字体大小
        cellFont.setFontName("宋体"); // 设置字体类型
        cellFont.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD); // 字体加粗
        cellstyle.setFont(cellFont); // 为表头样式设置字体样式


        for (int i = 0; i < fieldName.size(); i++) {
            HSSFCell cell = row.createCell((short) i);
            cell.setCellValue((String) fieldName.get(i));
            cell.setCellStyle(cellstyle);
        }

        YkSr ykSr = null;

        // 第五步，写入实体数据 实际应用中这些数据从数据库得到，

        for (int i = 0; i < fieldData.size(); i++) {
            row = sheet.createRow(i + 1);
            ykSr = (YkSr) fieldData.get(i);
            // 第四步，创建单元格，并设置值
            String[] nameArry = {ykSr.getProjects(), ykSr.getMinorX(), String.valueOf(ykSr.getGroupA()), String.valueOf(ykSr.getGroupB()),
                    String.valueOf(ykSr.getGroupC()), String.valueOf(ykSr.getGroupD()), ykSr.getSpareA(), ykSr.getSpareB(),
                    String.valueOf( ykSr.getHj())};
            for (int j = 0; j < nameArry.length; j++) {
                if (nameArry[j] == null) {
                    row.createCell((short) j).setCellValue("");
                } else {
                    row.createCell((short) j).setCellValue(nameArry[j]);
                }

            }



        }
        //自适应列宽
        for (int i = 0; i < fieldName.size(); i++) {
            sheet.autoSizeColumn((short) i);
        }
        return wb;
    }

    //将信息写入输出流的方法。
    public void exportExcel(List fieldName, List fieldData, OutputStream os) throws Exception {
        HSSFWorkbook wb = createWorkbook(fieldName, fieldData);
        wb.write(os);
        os.close();
    }
}